package com.ddshuai.easy;

import java.util.Stack;

/**
 * 描述 新通教育
 *
 * @author ddshuai
 * @date 2018-12-10 15:26
 * <p>
 * 你现在是棒球比赛记录员。
 * 给定一个字符串列表，每个字符串可以是以下四种类型之一：
 * 1.整数（一轮的得分）：直接表示您在本轮中获得的积分数。
 * 2. "+"（一轮的得分）：表示本轮获得的得分是前两轮有效 回合得分的总和。
 * 3. "D"（一轮的得分）：表示本轮获得的得分是前一轮有效 回合得分的两倍。
 * 4. "C"（一个操作，这不是一个回合的分数）：表示您获得的最后一个有效 回合的分数是无效的，应该被移除。
 * <p>
 * 每一轮的操作都是永久性的，可能会对前一轮和后一轮产生影响。
 * 你需要返回你在所有回合中得分的总和。
 * <p>
 * 示例一
 * 输入: ["5","2","C","D","+"]
 * 输出: 30
 * 解释:
 * 第1轮：你可以得到5分。总和是：5。
 * 第2轮：你可以得到2分。总和是：7。
 * 操作1：第2轮的数据无效。总和是：5。
 * 第3轮：你可以得到10分（第2轮的数据已被删除）。总数是：15。
 * 第4轮：你可以得到5 + 10 = 15分。总数是：30。
 * <p>
 * 实例二
 * 输入: ["5","-2","4","C","D","9","+","+"]
 * 输出: 27
 * 解释:
 * 第1轮：你可以得到5分。总和是：5。
 * 第2轮：你可以得到-2分。总数是：3。
 * 第3轮：你可以得到4分。总和是：7。
 * 操作1：第3轮的数据无效。总数是：3。
 * 第4轮：你可以得到-4分（第三轮的数据已被删除）。总和是：-1。
 * 第5轮：你可以得到9分。总数是：8。
 * 第6轮：你可以得到-4 + 9 = 5分。总数是13。
 * 第7轮：你可以得到9 + 5 = 14分。总数是27。
 * <p>
 * 注意：
 * <p>
 * 输入列表的大小将介于1和1000之间。
 * 列表中的每个整数都将介于-30000和30000之间。
 **/
public class CalPoints {

    /**
     * 自己的解答
     * @param ops
     * @return
     */
    public int calPoints(String[] ops) {
        final String C = "C";
        final String D = "D";
        final String ADD = "+";
        int count = 0;
        //操作栈
        Stack<Integer> all = new Stack<>();
        for (String op : ops) {
           if (C.equals(op)){
               Integer del = all.pop();
               count -= del;
           }else if (ADD.equals(op)){
               Integer first = all.pop();
               Integer second = all.pop();
               Integer sum = first+second;
               all.push(second);
               all.push(first);
               all.push(sum);
               count += sum;
           }else if (D.equals(op)){
               Integer pre = all.pop();
               int cur = pre << 1;
               all.push(pre);
               all.push(cur);
               count += cur;
           } else {
               //直接压入栈中
               all.push(Integer.parseInt(op));
               count += Integer.parseInt(op);
           }
        }
        return count;
    }

    /**
     * 效率最高的
     * @param ops
     * @return
     */
    public int calPoints2(String[] ops){
        int[] scores = new int[2000];
        int ct=-1;
        for(int i=0;i<ops.length;i++){
            if(ops[i].equals("+")){
                scores[ct+1]=scores[ct]+scores[ct-1];
                ct++;
            }else if(ops[i].equals("D")){
                scores[ct+1]=scores[ct]*2;
                ct++;
            }else if(ops[i].equals("C")){
                ct--;
            }else{
                scores[++ct]=Integer.parseInt(ops[i]);
            }
        }
        int num=0;

        for(int i=0;i<=ct;i++){
            num+=scores[i];
        }
        return num;
    }

    /**
     * 官方解答
     * @param ops
     * @return
     */
    public int calPoints3(String[] ops) {
        Stack<Integer> stack = new Stack();

        for(String op : ops) {
            if (op.equals("+")) {
                int top = stack.pop();
                int newtop = top + stack.peek();
                stack.push(top);
                stack.push(newtop);
            } else if (op.equals("C")) {
                stack.pop();
            } else if (op.equals("D")) {
                stack.push(2 * stack.peek());
            } else {
                stack.push(Integer.valueOf(op));
            }
        }

        int ans = 0;
        for(int score : stack) ans += score;
        return ans;
    }


    public static void main(String[] args) {
        //String[] strings = new String[]{"5","-2","4","C","D","9","+","+"};
        String[] strings = new String[]{"5","2","C","D","+"};
        System.out.println(new CalPoints().calPoints(strings));
    }
}
